#ifndef _HASH_ENCODE_H
#define _HASH_ENCODE_H

#include <torch/torch.h>

extern "C" __declspec(dllexport) void hash_encoder_forward(const at::Tensor* inputs, const at::Tensor* embeddings, const at::Tensor* offsets,
	const at::Tensor* box_min, const at::Tensor* box_max, const double base_resolution, const double b, const at::Tensor* outputs, const at::Tensor* dy_dx);

extern "C" __declspec(dllexport) void hash_encoder_backward(const at::Tensor* grad, const at::Tensor* inputs, const at::Tensor* embeddings, const at::Tensor* offsets,
	const at::Tensor* box_min, const at::Tensor* box_max, const double base_resolution, const double b, const at::Tensor* grad_embeddings);

#endif